#pragma once
#include <torch/torch.h>
#include <string>

class Denoiser {
public:
    Denoiser(const std::string& model_path, const std::string& device = "cpu");
    torch::Tensor denoise(const torch::Tensor& audio);
    void process_file(const std::string& input_path, const std::string& output_path);
private:
    torch::jit::Module model_;
    torch::Device device_;
    int sample_rate_ = 16000;  // 默认采样率
}; 